/*
 * Tencent is pleased to support the open source community by making BK-CI 蓝鲸持续集成平台 available.
 *
 * Copyright (C) 2019 THL A29 Limited, a Tencent company.  All rights reserved.
 *
 * BK-CI 蓝鲸持续集成平台 is licensed under the MIT license.
 *
 * A copy of the MIT License is included in this file.
 *
 *
 * Terms of the MIT License:
 * ---------------------------------------------------
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
 * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of
 * the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
 * LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
 * NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

package com.tencent.bkrepo.fs.server.storage

import com.tencent.bkrepo.common.artifact.api.ArtifactFile
import com.tencent.bkrepo.common.artifact.stream.ArtifactInputStream
import com.tencent.bkrepo.common.artifact.stream.Range
import com.tencent.bkrepo.common.storage.core.StorageService
import com.tencent.bkrepo.common.storage.credentials.StorageCredentials
import com.tencent.bkrepo.common.storage.pojo.RegionResource
import com.tencent.bkrepo.fs.server.RepositoryCache
import com.tencent.bkrepo.fs.server.model.TBlockNode
import com.tencent.bkrepo.fs.server.service.BlockNodeService
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.withContext

class CoStorageManager(
    private val blockNodeService: BlockNodeService,
    private val storageService: StorageService
) {

    suspend fun storeBlock(artifactFile: ArtifactFile, blockNode: TBlockNode) {
        withContext(Dispatchers.IO) {
            val digest = artifactFile.getFileSha256()
            val repo = RepositoryCache.getRepoDetail(blockNode.projectId, blockNode.repoName)
            val storageCredentials = repo.storageCredentials
            val stored = storageService.store(digest, artifactFile, storageCredentials)
            try {
                blockNodeService.createBlock(blockNode, storageCredentials)
            } catch (e: Exception) {
                if (stored > 1) {
                    storageService.delete(digest, storageCredentials)
                }
                throw e
            }
        }
    }

    suspend fun loadArtifactInputStream(
        blocks: List<RegionResource>,
        range: Range,
        storageCredentials: StorageCredentials?
    ): ArtifactInputStream? {
        return withContext(Dispatchers.IO) {
            storageService.load(blocks, range, storageCredentials)
        }
    }
}
